package com.zzyl.common.utils.ollama.httpUtil;

import cn.hutool.http.HttpUtil;
import cn.hutool.json.JSONObject;

public class OllamaHttpUtil {
    public static String REQUEST_URL = "http://192.168.34.203:11434/api/generate";
    public static String MODEL = "mistral";

    /**
     * 向本地 Ollama 模型发送文本生成请求。
     *
     * @param userInput 用户输入内容
     * @param systemPrompt 系统提示或上下文
     * @return 模型返回的文本响应
     */
    public static String sendGenerator(String userInput, String systemPrompt) {
        try {
            JSONObject entries = new JSONObject();
            entries.set("prompt", systemPrompt + "\n用户输入如下:\n" + userInput);
            entries.set("model", MODEL);
            entries.set("stream", false);
            entries.set("temperature", 0.2);
            entries.set("top_p", 0.7);

            String post = HttpUtil.post(REQUEST_URL, entries.toString());
            JSONObject json = new JSONObject(post);
            return json.getStr("response");
        } catch (Exception e) {
            return "调用失败：" + e.getMessage();
        }
    }
}
